package com.zy.explore.license.interfaces.assembler;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.zy.explore.license.application.service.StationMsgService;
import com.zy.explore.license.application.service.UserMsgService;
import com.zy.explore.license.domain.mybatis.entity.StationMsg;
import com.zy.explore.license.domain.mybatis.entity.UserMsg;
import com.zy.explore.license.infrastructure.util.DataBeanUtil;
import com.zy.explore.license.infrastructure.util.JwtUtil;
import com.zy.explore.license.interfaces.frm.CommonIdFrm;
import com.zy.explore.license.interfaces.frm.UserMsgFrm;
import com.zy.explore.license.interfaces.vo.PageVO;
import com.zy.explore.license.interfaces.vo.UserMsgVO;
import com.google.common.collect.Lists;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
import org.springframework.data.domain.PageRequest;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * 用户消息Assembler
 *
 * @author ChinaYin  主页:https://gitee.com/zy-explorej2ee 邮箱:zyexplorej2ee@sina.com
 * @date 2021年06月13日 12:13:50
 */
@Component
public class UserMsgAssembler {

    /**
     * 注入service
     */
    @Resource
    private UserMsgService service;

    @Resource
    private StationMsgService stationMsgService;

    /**
     * 新增数据
     *
     * @param frm 参数对象
     * @return 视图对象
     */
    public UserMsgVO insert(UserMsgFrm frm) {
        UserMsg entity = convertFromFrm(frm);
        entity.setCreateTime(new Date());
        boolean res = service.save(entity);
        if (res) {
            return convertToVO(entity);
        }
        return null;
    }

    /**
     * 删除对象
     *
     * @param frm 参数对象
     * @return true:删除成功 false:删除失败
     */
    public boolean delete(CommonIdFrm frm) {
        return service.removeById(Long.parseLong(frm.getId()));
    }

    /**
     * 更新对象
     *
     * @param frm 待更新的Frm对象
     * @return 成功返回更新后的VO对象，失败返回null
     */
    public UserMsgVO update(UserMsgFrm frm) {
        UserMsg entity = convertFromFrm(frm);
        entity.setUpdateTime(new Date());
        boolean res = service.updateById(entity);
        if (res) {
            return convertToVO(entity);
        }
        return null;
    }

    /**
     * 根据ID查询数据
     *
     * @param frm 主键Frm对象
     * @return 成功返回对应ID的VO对象，失败返回null
     */
    public UserMsgVO findById(CommonIdFrm frm) {
        UserMsg entity = service.getById(Long.parseLong(frm.getId()));
        return convertToVO(entity);
    }

    /**
     * 根据条件查询数据集合
     *
     * @param frm 查询条件对象
     * @return 满足条件的数据集合
     */
    public List<UserMsgVO> getList(UserMsgFrm frm) {
        // 查询条件
        QueryWrapper<UserMsg> wrapper = UserMsg.getWrapper(frm);
        // 设置排序
        UserMsg.setSortCriteria(frm.getSortFields(), wrapper, null, null);
        List<UserMsg> list = UserMsg.builder().build().findList(wrapper);
        if (!CollectionUtils.isEmpty(list)) {
            List<UserMsgVO> retList = new ArrayList<>();
            list.forEach(x -> {
                UserMsgVO vo = convertToVO(x);
                StationMsg stationMsg = stationMsgService.getById(x.getMsgId());
                vo.setContent(stationMsg.getMsgContent());
                vo.setTitle(stationMsg.getTitle());
                retList.add(vo);
            });
            return retList;
        }
        return new ArrayList<>();
    }

    /**
     * 根据条件查询分页
     *
     * @param frm 查询条件
     * @return 满足条件的分页
     */
    public PageVO<UserMsgVO> getPageInfo(UserMsgFrm frm) {
        // 设置条件
        QueryWrapper<UserMsg> wrapper = UserMsg.getWrapper(frm);
        // 设置排序
        UserMsg.setSortCriteria(frm.getSortFields(), wrapper, null, null);
        // 分页参数
        int pageNum = frm.getPageNum();
        int pageSize = frm.getPageSize();
        int total = 0;
        PageImpl<?> pageImpl = new PageImpl<>(Lists.newArrayList(), PageRequest.of(pageNum - 1, pageSize), total);
        // 查询分页
        Page<UserMsg> pageInfo = UserMsg.builder().build().findPageInfo(pageImpl, wrapper);
        // 转成PageVO
        PageRequest pageRequest = PageRequest.of(pageNum - 1, pageSize);
        List<UserMsgVO> retList = new ArrayList<>();
        Page<UserMsgVO> retPage = new PageImpl<>(retList, pageRequest, 0L);
        if (pageInfo == null || CollectionUtils.isEmpty(pageInfo.getContent())) {
            return new PageVO<>(retPage);
        }
        pageInfo.getContent().forEach(x -> retList.add(convertToVO(x)));
        retPage = new PageImpl<>(retList, pageRequest, pageInfo.getTotalElements());
        return new PageVO<>(retPage);
    }

    /**
     * 逻辑删除对象
     *
     * @param dto 参数对象
     * @return 1:删除成功 其他:删除失败
     */
    public boolean logicDelete(CommonIdFrm dto) {
        return service.updateById(UserMsg.builder().id(Long.parseLong(dto.getId())).delFlag(1).build());
    }

    /**
     * 获取用户未读的消息数量
     *
     * @param frm 查询参数对象
     * @return 返回未读消息数量
     */
    public int getUnReadMsgNum(UserMsgFrm frm) {
        frm.setUserId(JwtUtil.getCurrentUserId());
        frm.setReadStatus(0);
        List<UserMsgVO> list = this.getList(frm);
        if (CollectionUtils.isEmpty(list)) {
            return 0;
        }
        return list.size();
    }

    /**
     * 标记所有的消息为已读
     *
     * @param frm 入参对象
     * @return 返回更新的数量
     */
    public boolean readAll(UserMsgFrm frm) {
        QueryWrapper<UserMsg> wrapper = new QueryWrapper<>();
        wrapper.eq("read_status", frm.getReadStatus());
        wrapper.eq("update_time", new Date());
        return service.update(UserMsg.builder().userId(JwtUtil.getCurrentUserId()).build(), wrapper);
    }

    /**
     * 逻辑删除全部站内信
     *
     * @return 返回删除的数量
     */
    public boolean logicDeleteAll() {
        // TODO
        return true;
    }

    /**
     * 物理删除全部站内信
     *
     * @return 返回删除的数量
     */
    public boolean deleteAll() {
        // TODO
        return true;
    }

    /**
     * frm转实体类
     *
     * @param frm 参数对象
     * @return 转换后的实体类对象
     */
    public static UserMsg convertFromFrm(UserMsgFrm frm) {
        return DataBeanUtil.copyOf(frm, UserMsg.class, true);
    }

    /**
     * 实体类转VO
     *
     * @param entity 实体类
     * @return 转换后的对象
     */
    public static UserMsgVO convertToVO(UserMsg entity) {
       return entity == null ? null : DataBeanUtil.copyOf(entity, UserMsgVO.class, true);
    }
}
